package org.codefinger.dao.antlr.v4.runtime.tree.xpath;

import org.codefinger.dao.antlr.v4.runtime.tree.ParseTree;
import org.codefinger.dao.antlr.v4.runtime.tree.TerminalNode;
import org.codefinger.dao.antlr.v4.runtime.tree.Tree;
import org.codefinger.dao.antlr.v4.runtime.tree.Trees;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

public class XPathTokenElement extends XPathElement {
	protected int	tokenType;

	public XPathTokenElement(String tokenName, int tokenType) {
		super(tokenName);
		this.tokenType = tokenType;
	}

	@Override
	public Collection<ParseTree> evaluate(ParseTree t) {
		// return all children of t that match nodeName
		List<ParseTree> nodes = new ArrayList<ParseTree>();
		for (Tree c : Trees.getChildren(t)) {
			if (c instanceof TerminalNode) {
				TerminalNode tnode = (TerminalNode) c;
				if ((tnode.getSymbol().getType() == tokenType && !invert) || (tnode.getSymbol().getType() != tokenType && invert)) {
					nodes.add(tnode);
				}
			}
		}
		return nodes;
	}
}
